package com.qd.panda.service.wallet.publish;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.db.BaseService;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.DateUtil;
import com.cdqidi.util.RedisTemplateUtil;
import com.qd.common.sys.enums.BaseTableEnum;
import com.qd.common.panda.domain.entity.wallet.publish.PlanPublishDTO;
import com.qd.common.panda.domain.entity.wallet.publish.PlanPublishListDTO;
import com.qd.panda.mapper.wallet.publish.PlanPublishMapper;
import com.qd.panda.model.PlanPublish;
import com.qd.panda.model.PlanPublishList;
import com.qd.system.service.dict.MbUtil;
import com.qd.system.service.login.LoginUserService;
import com.qd.system.service.org.OrgService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.time.LocalDate;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

import static java.util.stream.Collectors.toList;

/**
 * 发布套餐
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class PlanPublishService extends BaseService<PlanPublishMapper, PlanPublish> {

    private static final String PREFIX_ID = "panda_plan_publish_publishId_{0}";
    //保存启用的套餐，默认只能启用一个
    private static final String ENABLE = "panda_plan_publish";

    private final RedisTemplateUtil<PlanPublish> redisTemplateUtil;
    private final LoginUserService loginUserService;
    private final OrgService orgService;
    private final PlanPublishListService planPublishListService;

    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public PlanPublishDTO save(PlanPublishDTO dto) {
        validator(dto);
        final PlanPublish model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                saveChild(dto);
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public PlanPublishDTO update(PlanPublishDTO dto) {
        final PlanPublish historyModel = Optional.ofNullable(getById(dto.getPublishId())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final PlanPublish model = dtoToModel(dto);
        try {
            if (super.updateById(model)) {
                return modelToDto(model);
            }
            throw new ApiException("更新失败");
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
            model.freeData();
        }
    }

    /**
     * 批量添加
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void saveBatch(List<PlanPublish> mList) {
        try {
            if (!super.saveBatch(mList)) {
                throw new ApiException("批量添加失败");
            }
        } finally {
            clearCaches(mList);
        }
    }

    /**
     * 批量更新
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void updateBatchIds(List<PlanPublish> mList) {
        final List<String> ids = mList.stream().map(PlanPublish::getPublishId).collect(toList());
        final List<PlanPublish> historyModelList = checkIds(ids);
        try {
            if (!super.updateBatchById(mList)) {
                throw new ApiException("批量更新失败");
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
            ids.clear();
        }
    }

    /**
     * 删除，单个
     *
     * @param publishId 主键
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String publishId) {
        final PlanPublish historyModel = Optional.ofNullable(getById(publishId)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(publishId)) {
                planPublishListService.removeByPublishId(publishId);
                throw new ApiException("删除失败");
            }
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
        }
    }

    /**
     * 批量删除
     *
     * @param ids 主键集合
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeBatchIds(List<String> ids) {
        ids.forEach(this::removeById);
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(PlanPublishDTO dto) {
        final QueryWrapper<PlanPublish> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param publishId 主键
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String publishId) {
        boolean exists = false;
        if (StringUtils.hasLength(publishId)) {
            final PlanPublish model = getById(publishId);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param publishId 主键
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public PlanPublish getById(String publishId) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, publishId);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(publishId), PlanPublish.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param publishId 主键
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public PlanPublishDTO getByIdDto(String publishId) {
        return Optional.ofNullable(getById(publishId)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public PlanPublish getOne(PlanPublishDTO dto) {
        final QueryWrapper<PlanPublish> queryWrapper = createQueryWrapper(dto);
        PlanPublish model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public PlanPublishDTO getOneDto(PlanPublishDTO dto) {
        return Optional.ofNullable(getOne(dto)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据
     */
    @Transactional(readOnly = true)
    public List<PlanPublish> getList(PlanPublishDTO dto) {
        final QueryWrapper<PlanPublish> queryWrapper = createQueryWrapper(dto);
        final List<PlanPublish> list = super.list(queryWrapper);
        queryWrapper.clear();
        return list;
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据，转换成页面对象
     */
    @Transactional(readOnly = true)
    public List<PlanPublishDTO> getListDto(PlanPublishDTO dto) {
        return modelToDtoList(getList(dto));
    }

    /**
     * 分页数据
     *
     * @param page 分页对象
     * @param dto  查询参数
     * @return 分页信息
     */
    @Transactional(readOnly = true)
    public IPage<PlanPublishDTO> page(Page<PlanPublish> page, PlanPublishDTO dto) {
        final QueryWrapper<PlanPublish> queryWrapper = createQueryWrapper(dto);
        Page<PlanPublish> iPage = baseMapper.selectPage(page, queryWrapper);
        Optional.ofNullable(iPage.getRecords()).ifPresent(this::modelToDtoList);
        queryWrapper.clear();
        return iPage.convert(this::modelToDto);
    }

    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(PlanPublish model) {
        if (null == model) {
            return;
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getPublishId()));
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.buildKey(ENABLE, true));
    }

    /**
     * 批量清空缓存
     *
     * @param list 数据库对象
     */
    public void clearCaches(List<PlanPublish> list) {
        if (null == list || list.isEmpty()) {
            return;
        }
        List<String> ids = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getPublishId())).collect(toList());
        if (!ids.isEmpty()) {
            redisTemplateUtil.deleteByKey(ids);
            ids.clear();
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.buildKey(ENABLE, true));
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public PlanPublishDTO modelToDto(PlanPublish model) {
        if (null == model) {
            return null;
        }
        final PlanPublishDTO dto = new PlanPublishDTO();
        dto.setPublishId(model.getPublishId());
        dto.setTitle(model.getTitle());
        dto.setStartDate(DateUtil.localDateToStr(model.getStartDate()));
        dto.setEndDate(DateUtil.localDateToStr(model.getEndDate()));
        dto.setAddTime(model.getAddTime());
        dto.setUpdateTime(model.getUpdateTime());
        dto.setEnable(model.getEnable());
        dto.setOperator(model.getOperator());
        dto.setOperateOrgId(model.getOperateOrgId());
        buildDisplay(dto);
        return dto;
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public PlanPublish dtoToModel(PlanPublishDTO dto) {
        if (null == dto) {
            return null;
        }
        final PlanPublish model = new PlanPublish();
        model.setPublishId(dto.getPublishId());
        model.setTitle(dto.getTitle());
        model.setStartDate(DateUtil.strToLocalDate(dto.getStartDate()));
        model.setEndDate(DateUtil.strToLocalDate(dto.getEndDate()));
        model.setAddTime(dto.getAddTime());
        model.setUpdateTime(dto.getUpdateTime());
        model.setEnable(dto.getEnable());
        model.setOperator(dto.getOperator());
        model.setOperateOrgId(dto.getOperateOrgId());
        return model;
    }

    /**
     * 码表转换
     *
     * @param dto 页面对象
     */
    public void buildDisplay(PlanPublishDTO dto) {
        dto.setEnableDisplay(MbUtil.idToDisplay(dto.getEnable()));
        dto.setOperatorDisplay(loginUserService.idToDisplay(dto.getOperator()));
        dto.setOperateOrgIdDisplay(orgService.idToDisplay(dto.getOperateOrgId()));
        dto.setPlanPublishList(planPublishListService.getListByPublishIdDTO(dto.getPublishId()));
    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<PlanPublishDTO> modelToDtoList(List<PlanPublish> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<PlanPublish> dtoToModelList(List<PlanPublishDTO> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<PlanPublish> createQueryWrapper(PlanPublishDTO dto) {
        final QueryWrapper<PlanPublish> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(dto.getTitle())) {
                queryWrapper.like("title", dto.getTitle());
            }
            if (StringUtils.hasLength(dto.getEnable())) {
                queryWrapper.eq("enable", dto.getEnable());
            }
        }
        queryWrapper.orderByDesc("enable", "add_time");
        return queryWrapper;
    }

    /**
     * 批量检查主键在数据库中是否存在
     *
     * @param ids 主键集合
     * @return 主键查询出来的集合
     */
    public List<PlanPublish> checkIds(List<String> ids) {
        if (null == ids || ids.isEmpty()) {
            throw new ApiException("ids不能为空");
        }
        final List<PlanPublish> list = super.listByIds(ids);
        if (list.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(ids.toArray()));
        }
        if (ids.size() == list.size()) {
            return list;
        }
        List<String> dbIds = list.stream().map(PlanPublish::getPublishId).collect(toList());
        final List<String> notExistsList = ids.stream().filter(id -> !dbIds.contains(id)).collect(toList());
        dbIds.clear();
        if (!notExistsList.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(notExistsList.toArray()));
        }
        return list;
    }

    private void validator(PlanPublishDTO dto) {
        if (MbUtil.isNotExists(dto.getEnable(), BaseTableEnum.SF)) {
            throw new ApiException("是否启用值错误");
        }
        if (!StringUtils.hasLength(dto.getStartDate()) || !StringUtils.hasLength(dto.getEndDate())) {
            throw new ApiException("开始日期和结束日期要么全为空，要不必须全部填写");
        }
        if (StringUtils.hasLength(dto.getStartDate()) && StringUtils.hasLength(dto.getEndDate())) {
            LocalDate startDate = DateUtil.strToLocalDate(dto.getStartDate());
            LocalDate endDate = DateUtil.strToLocalDate(dto.getEndDate());
            if (startDate.isAfter(endDate)) {
                throw new ApiException("开始日期必须小于结束日期");
            }
        }
        if (dto.getEnable().equals(ConstantDto.SF1)) {
            checkEnable();
        }
    }

    private void checkEnable() {
        List<PlanPublish> list = baseMapper.selectList(new QueryWrapper<>());
        if (!list.isEmpty()) {
            List<PlanPublish> enableList = list.stream().filter(planPublish -> StringUtils.hasLength(planPublish.getEnable()) && planPublish.getEnable().equals(ConstantDto.SF1)).collect(toList());
            if (!enableList.isEmpty()) {
                throw new ApiException("只能启用一个");
            }
        }
    }

    private void saveChild(PlanPublishDTO dto) {
        List<PlanPublishListDTO> planPublishList = dto.getPlanPublishList();
        if (null != planPublishList && !planPublishList.isEmpty()) {
            planPublishList.forEach(pu -> pu.setPublishId(dto.getPublishId()));
            List<PlanPublishList> planPublishLists = planPublishListService.dtoToModelList(planPublishList);
            planPublishListService.saveBatch(planPublishLists);
            planPublishLists.clear();
            planPublishList.clear();
        }
    }

    @Transactional(rollbackFor = Exception.class)
    public void enable(PlanPublishDTO dto) {
        PlanPublishDTO planPublishDTO = getByIdDto(dto.getPublishId());
        if (null != planPublishDTO) {
            if (!planPublishDTO.getEnable().equals(dto.getEnable())) {
                if (dto.getEnable().equals(ConstantDto.SF1)) {
                    checkEnable();
                }
                planPublishDTO.setEnable(dto.getEnable());
                planPublishDTO.setUpdateTime(dto.getUpdateTime());
                planPublishDTO.setOperateOrgId(dto.getOperateOrgId());
                planPublishDTO.setOperator(dto.getOperator());
                update(planPublishDTO);
            }
            planPublishDTO.freeData();
        }
    }

    @Transactional(readOnly = true)
    public PlanPublishDTO getEnable() {
        String key = RedisTemplateUtil.buildKey(ENABLE, true);
        PlanPublish planPublish = redisTemplateUtil.get(new CacheKey().setKey(key), () -> {
            QueryWrapper<PlanPublish> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("enable", ConstantDto.SF1);
            List<PlanPublish> list = baseMapper.selectList(queryWrapper);
            queryWrapper.clear();
            if (null != list && !list.isEmpty()) {
                if (list.size() > 1) {
                    throw new ApiException("数据异常");
                }
                PlanPublish onePlanPublish = list.get(0);
                if (null != onePlanPublish) {
                    if (onePlanPublish.getEnable().equals(ConstantDto.SF0)) {
                        return null;
                    }
                    LocalDate now = LocalDate.now();
                    LocalDate startDate = onePlanPublish.getStartDate();
                    if (null != startDate) {
                        if (now.isBefore(startDate)) {
                            return null;
                        }
                    }
                    LocalDate endDate = onePlanPublish.getEndDate();
                    if (null != endDate) {
                        if (now.isAfter(endDate)) {
                            return null;
                        }
                    }
                    return onePlanPublish;
                }
            }
            return null;
        }, PlanPublish.class);
        return modelToDto(planPublish);
    }

    /**
     * 定期清理超过发布时间的发布套餐
     */
    @Transactional(rollbackFor = Exception.class)
    public void cleanExpiredPlan() {
        QueryWrapper<PlanPublish> queryWrapper = new QueryWrapper<>();
        queryWrapper.lt("end_date", DateUtil.getStringNowDate());
        List<PlanPublish> list = baseMapper.selectList(queryWrapper);
        queryWrapper.clear();
        if (!list.isEmpty()) {
            list.forEach(planPublish -> planPublish.setEnable(ConstantDto.SF0));
            updateBatchIds(list);
            list.clear();
        }
    }

    @Transactional(rollbackFor = Exception.class)
    public void updatePlanPublishList(List<PlanPublishList> planPublishLists) {
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.buildKey(ENABLE, true));
        planPublishListService.updateBatchIds(planPublishLists);
    }
}